package com.qianbao.qianbaobusiness.ui.widget;

import android.content.Context;
import android.util.AttributeSet;
import android.view.View;
import android.widget.RadioGroup;

import com.androidbase.utils.DisplayUtils;

public class FlowRadioGroup extends RadioGroup {
    public FlowRadioGroup(Context context) {
        super(context);
    }

    public FlowRadioGroup(Context context, AttributeSet attrs) {
        super(context, attrs);
    }

    @Override
    protected void onMeasure(int widthMeasureSpec, int heightMeasureSpec) {
        int maxWidth = MeasureSpec.getSize(widthMeasureSpec);
        int childCount = getChildCount();
        int x = 0;
        int y = 0;
        int row = 0;

        for (int index = 0; index < childCount; index++) {
            final View child = getChildAt(index);
            if (child.getVisibility() != View.GONE) {
                child.measure(0, 0);
                //                final int childWidthMeasureSpec = MeasureSpec.makeMeasureSpec(
                //                        0, MeasureSpec.EXACTLY);
                ////                final int childHeightMeasureSpec = getChildMeasureSpec(heightMeasureSpec,
                ////                        mPaddingTop + mPaddingBottom + lp.topMargin + lp.bottomMargin,
                ////                        lp.height);
                //
                //                child.measure(childWidthMeasureSpec, MeasureSpec.UNSPECIFIED);
                child.measure(MeasureSpec.UNSPECIFIED, MeasureSpec.UNSPECIFIED);
                // 此处增加onlayout中的换行判断，用于计算所需的高度
                int width = child.getMeasuredWidth();
                int height = child.getMeasuredHeight();
                int marginx = 0;
                x += width + marginx;
                //                y = row * height + height;
                if (x > maxWidth) {
                    x = width;
                    marginx = 0;
                    row++;
                }
                y = row * height + height + row * DisplayUtils.dp2px(8);
                marginx = ((MarginLayoutParams) child.getLayoutParams()).rightMargin;
            }
        }
        // 设置容器所需的宽度和高度
        setMeasuredDimension(maxWidth, y);
    }

    @Override
    protected void onLayout(boolean changed, int l, int t, int r, int b) {
        final int childCount = getChildCount();
        int maxWidth = r - l;
        int x = 0;
        int y = 0;
        int marginx = 0;
        int row = 0;
        for (int i = 0; i < childCount; i++) {
            final View child = this.getChildAt(i);
            if (child.getVisibility() != View.GONE) {
                int width = child.getMeasuredWidth();
                int height = child.getMeasuredHeight();
                x += width + marginx;
                //                y = row * height + height;
                if (x > maxWidth) {
                    x = width;
                    marginx = 0;
                    row++;

                }
                y = row * height + height + row * DisplayUtils.dp2px(8);
                //                child.layout(x - width + marginx, y - height, x + marginx, y);
                child.layout(x - width, y - height, x, y);
            }
            marginx = ((MarginLayoutParams) child.getLayoutParams()).rightMargin;
        }
    }
}
